原始题目:剑指 Offer 54. 二叉搜索树的第k大节点 (opens new window)
解题思路:
基于二叉搜索树的特点,可以对二叉搜索树进行中序遍历,然后将第 个节点的值返回即可。
代码:
public int kthLargest(TreeNode root, int k) {
Deque<TreeNode> queue = new LinkedList<>();
if (root != null) {
queue.push(root);
}
int ans = 0;
while (root != null || !queue.isEmpty()) {
while (root != null) {
queue.push(root);
root = root.right;
}
root = queue.pop();
if (--k == 0) {
ans = root.val;
break;
}
root = root.left;
}
return ans;
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
复杂度分析
- 时间复杂度: 是二叉搜索树的节点数。最差情况下,,需要遍历所有的节点。
- 空间复杂度:最差情况下,树退化成链表,需要存储所有的节点。